package com.shujia.streaming

import org.apache.spark.{SparkConf, SparkContext}
import org.apache.spark.streaming.{Durations, StreamingContext}
import org.apache.spark.streaming.dstream.DStream

object Demo03Window {
  def main(args: Array[String]): Unit = {
    /**
     * 热门微博：
     * 每隔10分钟，统计最近24小时的热门微博
     *
     * 简化：
     * 每隔5s，统计最近5s内的单词数量
     * 每隔5s，统计最近15s内的单词数量
     *
     * 需要使用滑动窗口
     *
     */

    val conf = new SparkConf()
    conf.setAppName("Demo03Window")
    conf.setMaster("local[2]")
    val sc = new SparkContext(conf)

    val ssc = new StreamingContext(sc, Durations.seconds(5))

    // 当对窗口进行优化时 需要用到之前的计算结果 所以需要设置checkpoint地址来保存计算结果
    ssc.checkpoint("Spark/data/checkpoint")

    val lineDS: DStream[String] = ssc.socketTextStream("master", 8888)

    // 每隔5s，统计最近15s内的单词数量

    /**
     * 如果滑动时间=间隔时间 ---> 滚动窗口
     * 如果滑动时间<间隔时间 ---> 滑动窗口
     */
    lineDS
      .flatMap(_.split(","))
      .map(word => (word, 1))
      //      .reduceByKeyAndWindow(_ + _, Durations.seconds(15))
      .reduceByKeyAndWindow(_ + _, _ - _, Durations.seconds(15))
      .print()


    ssc.start()
    ssc.awaitTermination()
    ssc.stop()

  }

}
